package com.hao.tool.web.common;

import cn.hutool.core.io.FileUtil;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.hao.tool.web.module.fileInfo.entity.TemplateInfo;

import java.io.File;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

/**
 * description
 *
 * @author RuiHaoZ on GZ.cmhk
 * @date 2019-11-27 22:15
 */
public class TemplateInfoUtil {

    public static List<TemplateInfo> getList(String basePath) {
        List<File> fileList = FileUtil.loopFiles(basePath);
        List<TemplateInfo> templateInfos = new ArrayList<>();
        fileList.forEach(file -> {
            if (file.getName().contains(".html")) {
                TemplateInfo templateInfo = new TemplateInfo();
                templateInfo.setId(IdWorker.getIdStr());
                templateInfo.setClickCount(0L);
                templateInfo.setCreateTime(LocalDateTime.now());
                templateInfo.setSource("17sucai");
                templateInfo.setCreateUser("RuiHaoZ");
                templateInfo.setModName("模板名称");
                templateInfo.setModType("html");
                templateInfo.setModPath(file.getAbsolutePath());
                templateInfos.add(templateInfo);
            }
        });
        templateInfos.forEach(templateInfo -> {
            String content = FileUtil.readUtf8String(templateInfo.getAccessUrl());
            int be = content.indexOf("<title>");
            int end = content.lastIndexOf("</title>");
            if (be > 0 && end > 0) {
                String modelName = content.substring(be + "<title>".length(), end);
                System.out.println(modelName);
                templateInfo.setModName(modelName);
            }

        });
        return templateInfos;
    }
}
